Fix missing declaration bug