As suggested by Ewan N, adding the two lines that make life a bit easier