quick and dirty test fix