added suggested test and code changed to pass it