Remove workaround for pytest issue