Updated tests to work on drone