added test
fixed broken test