Adding back in accidentally removed test