Fixed tests for python3