minor fix for adityas tests