Non-Markovianity in a non-thermal bath