Test fixing

Version 0.3.1