Fixes python bug