Changed code in post to match repo