Fixed test and implementation
Extra assert