Fix tests

Hi
I am no ruby developer and not used to its syntax , but I tried to run tests and they failed here. After fixing these two files they work here.