Their is another PR gh-512 about dtype handling. @vdumoulin do this affect the line above?