Change code to make it build in release