fix failing test after moving classes