Yes, the assert test needs to be moved too for things to work.