Tidied one code snippet.