Fix the code to use the specifc version that is working