Fixed issues on docker testing

Two more lines of code that had similar issues as the last commit fixed