Fix If statement for rostest