added fix for last test